package com.windchime.blog.utils;

import com.aliyun.oss.OSS;
import com.aliyun.oss.OSSClientBuilder;
import com.aliyun.oss.model.PutObjectRequest;
import org.joda.time.DateTime;
import org.springframework.web.multipart.MultipartFile;

import java.io.InputStream;
import java.util.UUID;

public class UploadUtil {

    public static String upload(MultipartFile file, String fileName) {
        // 工具类获取值
        String endpoint = AlyYmlUtils.END_POINT;
        String accessKeyId = AlyYmlUtils.KEY_ID;
        String accessKeySecret = AlyYmlUtils.KEY_SECRET;
        String bucketName = AlyYmlUtils.BUCKET_NAME;

        try {
            // 创建OSSClient实例。
            OSS ossClient = new OSSClientBuilder().build(endpoint, accessKeyId, accessKeySecret);

            // 获取上传文件输入流
            InputStream inputStream = file.getInputStream();

            // 获取文件名称
            String originalFilename = file.getOriginalFilename();

            // 1、在文件名称里面添加随机唯一的值
            String uuid = UUID.randomUUID().toString().replaceAll("-", "");
            originalFilename = uuid + originalFilename;

            // 2、把文件按照日期进行分类
            // 获取当前日期
            String datePath = new DateTime().toString("yyyy-MM/dd");
            // 拼接
            originalFilename = fileName + "/" + datePath + "/" + originalFilename;

            // 创建PutObjectRequest对象。
            /*
               第一个参数 Bucket名称
               第二个参数 上传到oss文件路径和文件名称
               第三个参数 上传文件输入流
             */
            PutObjectRequest putObjectRequest = new PutObjectRequest(bucketName, originalFilename, inputStream);

            // 上传文件
            ossClient.putObject(putObjectRequest);

            // 关闭OSSClient。
            ossClient.shutdown();

            // 把文件上传之后文件路径返回
            // 需要把上传到阿里云oss路径手动拼接出来
            // https://windchime.oss-cn-shenzhen.aliyuncs.com/003.jpg
            return "https://" + bucketName + "." + endpoint + "/" + originalFilename;
        } catch (Exception e) {
            e.printStackTrace();
            return null;
        }
    }
}
